# -*- coding: UTF-8 -*-

from DetectorDeAumentativos import *
from Tokenizador import *
import unittest

class detectorDeAumentativosTest(unittest.TestCase):
	
	def setUp(self):
		self.detector = DetectorDeAumentativos()
		self.tokenizador = Tokenizador()

	def test_publicable(self):
		text = 'De golpe y porrazo me moría de hambre.'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Aumentativo', []))

	def test_detectarAumentativo(self):
		text = 'forrazo'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Aumentativo', [('forrazo', 'forro')]))
		text = 'Ah pero sos un taradazo!'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Aumentativo', [('taradazo', 'tarado')]))
		text = 'tonton tenias que ser'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Aumentativo', [('tonton', 'tonto')]))
		text = 'No, pero si es boludote de nacimientote'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Aumentativo', [('boludote', 'boludo')]))	

if __name__ == "__main__": 
	unittest.main()
